package com.thp.bigdata.zkdst;

import java.io.IOException;
import java.util.ArrayList;
import java.util.List;

import org.apache.zookeeper.KeeperException;
import org.apache.zookeeper.WatchedEvent;
import org.apache.zookeeper.Watcher;
import org.apache.zookeeper.ZooKeeper;

/**
 * 分布式系统的客户端
 * @author tommy
 *
 */
public class DistributeClient {

	private static final String connectString = "bd1:2181,bd2:2181,bd3:2181";
	// 超时时间
	private static final int sessionTimeout = 2000;
	
	private static final String parentNode = "/servers";
	// 注意: 加volatile的意义何在
	private volatile List<String> serverList;
	
	
	private ZooKeeper zk = null;
	
	
	/**
	 * 创建到zk的客户端连接
	 * @throws IOException
	 */
	public void getConnect() throws IOException {
		zk = new ZooKeeper(connectString, sessionTimeout, new Watcher() {
			@Override
			public void process(WatchedEvent event) {
				// 收到事件通知后的回调函数(应该是我们自己的事件处理逻辑)
				try {
					getServerList();  // 重新更新服务器列表,并且注册了监听 下次发生这种时间 还是会被调用
				} catch (KeeperException | InterruptedException e) {
					e.printStackTrace();
				}
			}
		});
	}
	
	/**
	 * 获取服务器信息列表
	 * @throws InterruptedException 
	 * @throws KeeperException 
	 */
	public void getServerList() throws KeeperException, InterruptedException {
		// 获取服务器子节点信息  并且对父节点进行监听   一旦发生了变化,那么process就会被调用
		List<String> children = zk.getChildren(parentNode, true);  // 需要监听父节点
		// 先创建一个局部的list来存服务器信息
		List<String> servers = new ArrayList<String>();
		
		for(String child : children) {
			// child 只是子节点的节点名    子节点不需要监听   只需要监听父节点  看子节点是否变化
			byte[] data = zk.getData(parentNode + "/" + child , false, null);
			servers.add(new String(data));
		}
		// 把servers赋值给成员变量serverList,以提供给业务线程使用
		serverList = servers;
		
		// 显示服务器列表
		System.out.println(serverList);
	}
	
	/**
	 * 业务功能
	 * @throws InterruptedException 
	 */
	public void handleBussiness() throws InterruptedException {
		System.out.println("client start working...");
		Thread.sleep(Long.MAX_VALUE);
	}
	
	public static void main(String[] args) throws IOException, KeeperException, InterruptedException {
		// 1.获取zookeeper连接
		DistributeClient client = new DistributeClient();
		client.getConnect();
		// 2.获取servers的子节点信息(并监听),从中获取服务器信息列表
		client.getServerList();
		// 3.业务线程启动
		client.handleBussiness();
	}
	
	
}
